Rethinking On-Policy Distillation of Large Language Models II: One Training Example
重新审视大语言模型在线策略蒸馏:单训练样本
🧭 On-Policy 蒸馏 · 数据效率 ★ 深度解读
Tsinghua University · University of Chinese Academy of Sciences · Northeastern University · University of Illinois Urbana-Champaign · Johns Hopkins University
一个 query 训练数百步即可恢复大部分 full-data OPD 增益,揭示 OPD 数据过剩、算法吸收是瓶颈
📌 工作参考价值(4 点)
- 工业 OPD 数据筛选可转向 state coverage 视角:不是收集更多 prompt,而是选语义多样、能诱导互补状态的少量 query(如 16 个/域即可匹配 full-data MOPD)。在电商 query 改写、生成式推荐、Agent 轨迹生成中,可先对 prompt/query 做语义聚类,取代表点训练,大幅降低标注与训练成本。
- 冷启动或无标注场景可借鉴 content-light 模板与离域日志:只要 teacher 能提供 token-level 监督,chat template + <think> 等空模板或历史对话也能驱动 OPD 接近真实 query 效果;可尝试用模板启动生成式推荐或 Agent 策略蒸馏。
- OPD 训练瓶颈在吸收率而非数据量:固定 states 仍需数百步收敛,说明加大 batch 或数据量未必加速;应优先优化步效率,如 batch reuse、trust region、按 token teacher signal 加权,减少训练步数。
- 多教师蒸馏可按域选少量语义代表 query,每域 16 条即可匹配全量,适合多业务线共享一个生成式推荐/Agent 模型时的数据配比与成本控制。